package pers.xiaojun.boot.common.util;

import cn.hutool.core.collection.CollUtil;
import com.google.common.collect.Lists;

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 集合类型工具类
 *
 * @author xiaojun
 * @since 2025-10-23
 */
public class CollectionUtils {

    /**
     * 转换列表元素（过滤空数据）
     *
     * @param source 源集合
     * @param mapper 转换函数
     * @param <S>    source 来源类型
     * @param <T>    target 目标类型
     * @return 转换后的集合（过滤空数据）
     */
    public static <S, T> List<T> convertList(Collection<S> source, Function<S, T> mapper) {
        if (CollUtil.isEmpty(source)) {
            return Collections.emptyList();
        }
        return source.stream().map(mapper).filter(Objects::nonNull).collect(Collectors.toList());
    }

    /**
     * 转换Set集合元素（过滤空数据）
     *
     * @param source 源集合
     * @param mapper 转换函数
     * @param <S>    source 来源类型
     * @param <T>    target 目标类型
     * @return 转换后的Set集合（过滤空数据）
     */
    public static <S, T> Set<T> convertSet(Collection<S> source, Function<S, T> mapper) {
        if (CollUtil.isEmpty(source)) {
            return Collections.emptySet();
        }
        return source.stream().map(mapper).filter(Objects::nonNull).collect(Collectors.toSet());
    }

}
